package com.ljy.my_study.leetcode.最长定差子序列;

import java.util.HashMap;
import java.util.Map;

/**
 * @author lijunying
 * @date 2021/11/5 9:12
 */
public class TestMain {

//    给你一个整数数组 arr 和一个整数 difference，请你找出并返回 arr 中最长等差子序列的长度，该子序列中相邻元素之间的差等于 difference 。
//
//    子序列 是指在不改变其余元素顺序的情况下，通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
//
//             
//
//    示例 1：
//
//    输入：arr = [1,2,3,4], difference = 1
//    输出：4
//    解释：最长的等差子序列是 [1,2,3,4]。
//    示例 2：
//
//    输入：arr = [1,3,5,7], difference = 1
//    输出：1
//    解释：最长的等差子序列是任意单个元素。
//    示例 3：
//
//    输入：arr = [1,5,7,8,5,3,4,2,1], difference = -2
//    输出：4
//    解释：最长的等差子序列是 [7,5,3,1]。
//             
//
//    提示：
//
//            1 <= arr.length <= 105
//            -104 <= arr[i], difference <= 104
//
//    来源：力扣（LeetCode）
//    链接：https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference
//    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

    public static void main(String[] args) {
        int[] arr={-4,1,-3,14,-10,13,-9,0,-3,-14,-11,3,1,13};
        int diff=8;
        System.out.println(new TestMain().longestSubsequence(arr,diff));
    }

    public int longestSubsequence(int[] arr, int difference) {
        int max=0;
        int len=arr.length;
        int[] dp=new int[len];
        for(int i=0;i<len;i++){
            if(arr[i]==difference){
                dp[i]=Math.max(dp[i],1);
                max=Math.max(dp[i],max);
            }
            for(int j=i+1;j<len;j++){
                if(arr[j]-arr[i]==difference){
                    if(dp[j]==0){
                        dp[j]=2;
                    }
                    dp[j]=Math.max(dp[j],dp[i]+1);
                    max=Math.max(dp[j],max);
                }
            }
        }
        return max==0?1:max;
    }

    /**
     * 动态规划。使用key-value形式保存当前数据对比difference有多少个等差数列中数量。
     * @param arr
     * @param difference
     * @return
     */
    public int longestSubsequence2(int[] arr, int difference) {
        int ans = 0;
        Map<Integer, Integer> dp = new HashMap<Integer, Integer>();
        for (int v : arr) {
            dp.put(v, dp.getOrDefault(v - difference, 0) + 1);
            ans = Math.max(ans, dp.get(v));
        }
        return ans;
    }
}
